<?php  if (!defined('BASEPATH')) exit('No direct script access allowed');
// +----------------------------------------------------------------------+
// | Treacle PHP Framework Version 0.10                                   |
// +----------------------------------------------------------------------+
// | Copyright (c) 2007 Treacle.cn All rights reserved.					  |
// +----------------------------------------------------------------------+
// | Licensed under the Apache License, Version 2.0 (the 'License');      |
// | you may not use this file except in compliance with the License.     |
// | You may obtain a copy of the License at                              |
// | http://www.apache.org/licenses/LICENSE-2.0                           |
// | Unless required by applicable law or agreed to in writing, software  |
// | distributed under the License is distributed on an 'AS IS' BASIS,    |
// | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or      |
// | implied. See the License for the specific language governing         |
// | permissions and limitations under the License.                       |
// +----------------------------------------------------------------------+
// | Author: BoPo <ibopo@126.com>										  |
// +----------------------------------------------------------------------+
/**
 * Treacle FTP class file
 *
 * @author BoPo <ibopo@126.com>
 * @link http://www.treacle.cn/
 * @copyright Copyright &copy; 2007 Treacle
 * @license http://www.treacle.cn/license/
 * @version $Id $
 * @package libraries
 */
/**
 * FTP Class
 *
 * @package libraries
 * @author BoPo <ibopo@126.com>
 * @since 0.10
 * @version	$Id $
 */
class TFTP
{
	private $hostname	= '';
	private $username	= '';
	private $password	= '';
	private $port		= 21;
	private $passive	= TRUE;
	private $debug		= FALSE;
	private $conn_id	= FALSE;
	/**
	 * Constructor - Sets Preferences
	 *
	 * The constructor can be passed an array of config values
	 */	
	public function __construct($config = array())
	{
		if (count($config) > 0)
		{
			$this->initialize($config);
		}
	}
	/**
	 * Initialize preferences
	 *
	 * @access	public
	 * @param	array
	 * @return	void
	 */	
	public function initialize($config = array())
	{
		foreach ($config as $key => $val)
		{
			if (isset($this->$key))
			{
				$this->$key = $val;
			}
		}
		$this->hostname = preg_replace('|.+?://|', '', $this->hostname);
	}
	/**
	 * FTP 连接操作
	 *
	 * @access	public
	 * @param	array	 FTP 连接参数
	 * @return	bool
	 */	
	public function connect($config = array())
	{
		if (count($config) > 0)
		{
			$this->initialize($config);
		}
		if (FALSE === ($this->conn_id = @ftp_connect($this->hostname, $this->port)))
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_connect');
			}
			return FALSE;
		}
		if ( ! $this->_login())
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_login');
			}
			return FALSE;
		}
		// Set passive mode if needed
		if ($this->passive == TRUE)
		{
			ftp_pasv($this->conn_id, TRUE);
		}
		return TRUE;
	}
	/**
	 * FTP 登录
	 *
	 * @access	private
	 * @return	bool
	 */	
	private function _login()
	{
		return @ftp_login($this->conn_id, $this->username, $this->password);
	}
	/**
	 * Validates the connection ID
	 *
	 * @access	private
	 * @return	bool
	 */	
	private function _is_conn()
	{
		if ( ! is_resource($this->conn_id))
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_no_connection');
			}
			return FALSE;
		}
		return TRUE;
	}
	/**
	 * 修改目录(cd)
	 *
	 * The second parameter lets us momentarily turn off debugging so that
	 * this public function can be used to test for the existance of a folder
	 * without throwing an error.  There's no FTP equivalent to is_dir()
	 * so we do it by trying to change to a particular directory.  
	 * Internally, this paramter is only used by the "mirror" public function below.
	 *
	 * @access	public
	 * @param	string
	 * @param	bool
	 * @return	bool
	 */	
	public function changedir($path = '', $supress_debug = FALSE)
	{
		if ($path == '' OR ! $this->_is_conn())
		{
			return FALSE;
		}
		$result = @ftp_chdir($this->conn_id, $path);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE AND $supress_debug == FALSE)
			{
				$this->_error('ftp_unable_to_changedir');
			}
			return FALSE;
		}
		return TRUE;
	}
	/**
	 * 创建一个目录(mkdir)
	 *
	 * @access	public
	 * @param	string
	 * @return	bool
	 */	
	public function mkdir($path = '', $permissions = NULL)
	{
		if ($path == '' OR ! $this->_is_conn())
		{
			return FALSE;
		}
		$result = @ftp_mkdir($this->conn_id, $path);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_makdir');
			}
			return FALSE;
		}
		// Set file permissions if needed
		if ( ! is_null($permissions))
		{
			$this->chmod($path, (int)$permissions);
		}
		return TRUE;
	}
	/**
	 * 上传文件到服务器(upload)
	 *
	 * @access	public
	 * @param	string
	 * @param	string
	 * @param	string
	 * @return	bool
	 */	
	public function upload($locpath, $rempath, $mode = 'auto', $permissions = NULL)
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		if ( ! file_exists($locpath))
		{
			$this->_error('ftp_no_source_file');
			return FALSE;
		}
		// Set the mode if not specified
		if ($mode == 'auto')
		{
			// Get the file extension so we can set the upload type
			$ext = $this->_getext($locpath);
			$mode = $this->_settype($ext);
		}
		$mode 	= ($mode == 'ascii') ? FTP_ASCII : FTP_BINARY;
		$result = @ftp_put($this->conn_id, $rempath, $locpath, $mode);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_upload');
			}
			return FALSE;
		}
		// Set file permissions if needed
		if ( ! is_null($permissions))
		{
			$this->chmod($rempath, (int)$permissions);
		}
		return TRUE;
	}
	/**
	 * 文件改名或者删除
	 *
	 * @access	public
	 * @param	string
	 * @param	string
	 * @param	bool
	 * @return	bool
	 */	
	public function rename($old_file, $new_file, $move = FALSE)
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		$result = @ftp_rename($this->conn_id, $old_file, $new_file);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE)
			{
				$msg = ($move = FALSE) ? 'ftp_unable_to_remame' : 'ftp_unable_to_move';
				$this->_error($msg);
			}
			return FALSE;
		}
		return TRUE;
	}
	/**
	 * 移动一个文件
	 *
	 * @access	public
	 * @param	string
	 * @param	string
	 * @return	bool
	 */	
	public function move($old_file, $new_file)
	{
		return $this->rename($old_file, $new_file, TRUE);
	}
	/**
	 * 改名或者移动一个文件
	 *
	 * @access	public
	 * @param	string
	 * @return	bool
	 */	
	public function delete_file($filepath)
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		$result = @ftp_delete($this->conn_id, $filepath);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_delete');
			}
			return FALSE;
		}
		return TRUE;
	}
	/**
	 * Delete a folder and recursively delete everything (including sub-folders)
	 * containted within it.
	 *
	 * @access	public
	 * @param	string
	 * @return	bool
	 */	
	public function delete_dir($filepath)
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		// Add a trailing slash to the file path if needed
		$filepath 	= preg_replace("/(.+?)\/*$/", "\\1/",  $filepath);
		$list	 	= $this->list_files($filepath);
		if ($list !== FALSE AND count($list) > 0)
		{
			foreach ($list as $item)
			{
				// If we can't delete the item it's probaly a folder so
				// we'll recursively call delete_dir()
				if ( ! @ftp_delete($this->conn_id, $filepath.$item))
				{
					$this->delete_dir($filepath.$item);
				}
			}
		}
		$result = @ftp_rmdir($this->conn_id, $filepath);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_delete');
			}
			return FALSE;
		}
		return TRUE;
	}
	/**
	 * Set file permissions
	 *
	 * @access	public
	 * @param	string 	the file path
	 * @param	string	the permissions
	 * @return	bool
	 */		
	public function chmod($path, $perm)
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		// Permissions can only be set when running PHP 5
		if ( ! function_exists('ftp_chmod'))
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_chmod');
			}
			return FALSE;
		}
		$result = @ftp_chmod($this->conn_id, $perm, $path);
		if ($result === FALSE)
		{
			if ($this->debug == TRUE)
			{
				$this->_error('ftp_unable_to_chmod');
			}
			return FALSE;
		}
		return TRUE;
	}
	/**
	 * 遍历一个目录的文件
	 *
	 * @access	public
	 * @return	array
	 */	
	public function list_files($path = '.')
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		return ftp_nlist($this->conn_id, $path);
	}
	/**
	 * Read a directory and recreate it remotely
	 *
	 * This public function recursively reads a folder and everything it contains (including
	 * sub-folders) and creates a mirror via FTP based on it.  Whatever the directory structure
	 * of the original file path will be recreated on the server.
	 *
	 * @access	public
	 * @param	string	path to source with trailing slash
	 * @param	string	path to destination - include the base folder with trailing slash
	 * @return	bool
	 */	
	public function mirror($locpath, $rempath)
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		// Open the local file path
		if ($fp = @opendir($locpath))
		{
			// Attempt to open the remote file path.
			if ( ! $this->changedir($rempath, TRUE))
			{
				// If it doesn't exist we'll attempt to create the direcotory
				if ( ! $this->mkdir($rempath) OR ! $this->changedir($rempath))
				{
					return FALSE;
				}
			}
			// Recursively read the local directory
			while (FALSE !== ($file = readdir($fp)))
			{
				if (@is_dir($locpath.$file) && substr($file, 0, 1) != '.')
				{
					$this->mirror($locpath.$file."/", $rempath.$file."/");
				}
				elseif (substr($file, 0, 1) != ".")
				{
					// Get the file extension so we can se the upload type
					$ext = $this->_getext($file);
					$mode = $this->_settype($ext);

					$this->upload($locpath.$file, $rempath.$file, $mode);
				}
			}
			return TRUE;
		}
		return FALSE;
	}
	/**
	 * Extract the file extension
	 *
	 * @access	private
	 * @param	string
	 * @return	string
	 */	
	private function _getext($filename)
	{
		if (FALSE === strpos($filename,'.'))
		{
			return 'txt';
		}
		$x = explode('.', $filename);
		return end($x);
	}
	/**
	 * 设置上传文件的类型
	 *
	 * @access	private
	 * @param	string
	 * @return	string
	 */	
	private function _settype($ext)
	{
		$text_types = array('txt','text','php','phps','php4','js','css','htm','html','phtml','shtml','log','xml');
		return (in_array($ext, $text_types)) ? 'ascii' : 'binary';
	}
	/**
	 * 关闭连接
	 *
	 * @access	public
	 * @param	string	path to source
	 * @param	string	path to destination
	 * @return	bool
	 */	
	public function close()
	{
		if ( ! $this->_is_conn())
		{
			return FALSE;
		}
		@ftp_close($this->conn_id);
	}
	/**
	 * 显示错误信息
	 *
	 * @access	private
	 * @param	string
	 * @return	bool
	 */	
	private function _error($line)
	{
		die($line);
	}
}
/*
+---------------------------------------------------------------------------
|   Last Modified By BoPo
|   Date: Sat Nov 10 13:56:00 CST 2007
+--------------------------------------------------------------------------
*/
?>